"""
数据库模块
包含PostgreSQL、Redis和ChromaDB的连接管理
"""
from app.db.database import (
    db_manager,
    get_db,
    Base,
    init_database,
    close_database,
    check_database_health,
)
from app.db.redis_client import (
    redis_manager,
    get_redis,
    get_cache,
    init_redis,
    close_redis,
    check_redis_health,
)
from app.db.vector_db import (
    vector_db_manager,
    init_vector_db,
    close_vector_db,
    check_vector_db_health,
)

__all__ = [
    # PostgreSQL
    "db_manager",
    "get_db",
    "Base",
    "init_database",
    "close_database",
    "check_database_health",
    # Redis
    "redis_manager",
    "get_redis",
    "get_cache",
    "init_redis",
    "close_redis",
    "check_redis_health",
    # Vector DB
    "vector_db_manager",
    "init_vector_db",
    "close_vector_db",
    "check_vector_db_health",
]
